﻿//  Copyright 2019 Google Inc. All Rights Reserved.
//
//  Licensed under the Apache License, Version 2.0 (the "License");
//  you may not use this file except in compliance with the License.
//  You may obtain a copy of the License at
//
//  http://www.apache.org/licenses/LICENSE-2.0
//
//  Unless required by applicable law or agreed to in writing, software
//  distributed under the License is distributed on an "AS IS" BASIS,
//  WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
//  See the License for the specific language governing permissions and
//  limitations under the License.

// NOTE: This file is a modified version of NdrParser.cs from OleViewDotNet
// https://github.com/tyranid/oleviewdotnet. It's been relicensed from GPLv3 by
// the original author James Forshaw to be used under the Apache License for this
// project.

using System;
using System.IO;
using NtCoreLib.Ndr.Formatter;
using NtCoreLib.Ndr.Parser;

namespace NtCoreLib.Ndr.Dce;

#pragma warning disable CS1591 // Missing XML comment for publicly visible type or member
[Serializable]
public sealed class NdrPointerTypeReference : NdrBaseTypeReference
{
    public NdrBaseTypeReference Type { get; private set; }
    public NdrPointerFlags Flags { get; }

    internal override NdrCorrelationDescriptor GetConformanceDescriptor()
    {
        return Type.GetConformanceDescriptor();
    }

    internal override NdrCorrelationDescriptor GetVarianceDescriptor()
    {
        return Type.GetVarianceDescriptor();
    }

    internal NdrPointerTypeReference(NdrBaseTypeReference type)
        : base(NdrFormatCharacter.FC_POINTER)
    {
        Type = type;
        Flags = NdrPointerFlags.FC_SIMPLE_POINTER;
    }

    internal NdrPointerTypeReference(NdrParseContext context, NdrFormatCharacter format, BinaryReader reader) : base(format)
    {
        Flags = (NdrPointerFlags)reader.ReadByte();
        if (Flags.HasFlagSet(NdrPointerFlags.FC_SIMPLE_POINTER))
        {
            Type = new NdrSimpleTypeReference(ReadFormat(reader));
        }
        else
        {
            Type = Read(context, ReadTypeOffset(reader));
        }
    }

    private protected override string FormatType(INdrFormatterContext context)
    {
        if (Type is NdrBaseArrayTypeReference)
        {
            return context.FormatType(Type);
        }
        else
        {
            bool is_unique = Format == NdrFormatCharacter.FC_UP;
            return $"{(is_unique ? context.FormatComment("unique") : "")}{context.FormatType(Type)}*";
        }
    }

    public override int GetSize()
    {
        return IntPtr.Size;
    }

    protected override void OnFixupLateBoundTypes()
    {
        Type = GetIndirectType(Type);
    }
}
#pragma warning restore 1591

